# Maintainer: Torsten Keßler <tpkessler at archlinux dot org>

pkgbase=rocm-hip-sdk
pkgname=(rocm-hip-sdk
         rocm-hip-libraries
         rocm-opencl-sdk
         rocm-hip-runtime
         rocm-language-runtime
         rocm-ml-sdk
         rocm-ml-libraries)
pkgver=5.7.1
pkgrel=2
arch=('loong64' 'x86_64')
license=('custom:None')
url='https://rocm.docs.amd.com/'

package_rocm-hip-sdk() {
    pkgdesc='Develop applications using HIP and libraries for AMD platforms'
    depends=('rocm-core'
             'rocm-hip-libraries'
             'rocm-llvm'
             'rocm-hip-runtime'
             'hipblas'
             'hipcub'
             'hipfft'
             'hipsparse'
             'hipsolver'
             'miopen-hip'
             'rccl'
             'rocalution'
             'rocblas'
             'rocfft'
             'rocprim'
             'rocrand'
             'rocsolver'
             'rocsparse'
             'rocthrust')
}

package_rocm-hip-libraries() {
    pkgdesc='Develop certain applications using HIP and libraries for AMD platforms'
    depends=('rocm-core'
             'rocm-hip-runtime'
             'hipblas'
             'hipfft'
             'hipsparse'
             'hipsolver'
             'rccl'
             'rocalution'
             'rocblas'
             'rocfft'
             'rocrand'
             'rocsolver'
             'rocsparse')
}

package_rocm-opencl-sdk() {
    pkgdesc='Develop OpenCL-based applications for AMD platforms'
    depends=('rocm-core'
             'hsa-rocr'
             'rocm-opencl-runtime'
             'hsakmt-roct')
}

package_rocm-hip-runtime() {
    pkgdesc='Packages to run HIP applications on the AMD platform'
    depends=('rocm-core'
             'rocm-language-runtime'
             'rocminfo'
             'hip-runtime-amd'
             'rocm-llvm'
             'rocm-cmake')
}

package_rocm-language-runtime() {
    pkgdesc='ROCm runtime'
    depends=('rocm-core'
             'hsakmt-roct'
             'hsa-rocr'
             'rocm-device-libs'
             'comgr')
}

package_rocm-ml-sdk() {
    pkgdesc='develop and run Machine Learning applications optimized for AMD platforms'
    depends=('rocm-core'
             'rocm-hip-sdk'
             'rocm-ml-libraries'
             'miopen-hip')
}

package_rocm-ml-libraries() {
    pkgdesc='Packages for key Machine Learning libraries'
    depends=('rocm-core'
             'rocm-hip-libraries'
             'rocm-llvm'
             'miopen-hip')
}
